package com.heima.media.service.impl;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.heima.article.dto.ArticleDto;
import com.heima.article.feign.ArticleFeign;
import com.heima.common.aliyun.service.AliyunGreenService;
import com.heima.common.dto.ResponseResult;
import com.heima.common.enums.AppHttpCodeEnum;
import com.heima.common.enums.MediaEnum;
import com.heima.common.minio.service.MinIOService;
import com.heima.common.util.SensitiveWordUtil;
import com.heima.media.dto.ContentDto;
import com.heima.media.entity.WmChannel;
import com.heima.media.entity.WmNews;
import com.heima.media.entity.WmSensitive;
import com.heima.media.entity.WmUser;
import com.heima.media.service.*;
import io.seata.core.context.RootContext;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringUtils;
import org.redisson.api.RBlockingDeque;
import org.redisson.api.RDelayedQueue;
import org.redisson.api.RedissonClient;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.io.InputStream;
import java.util.*;
import java.util.concurrent.TimeUnit;
import java.util.stream.Collectors;

/**
 * 将字体媒体提审文章进行送审业务
 */
@Service
public class AuditServiceImpl implements IAuditService {


    @Autowired
    private IWmNewsService wmNewsService;

    @Autowired
    private ArticleFeign articleFeign;

    @Autowired
    private IWmUserService wmUserService;

    @Autowired
    private IWmChannelService channelService;


    @Autowired
    private MinIOService minIOService;


    @Autowired
    private AliyunGreenService aliyunGreenService;


    @Autowired
    private RedissonClient redissonClient;


    /**
     * 根据提交的提审文章ID（自媒体文章ID）进行对指定自媒体文章进行审核
     *
     * @param id 自媒体文章ID
     */
    @Override
    public void audit(Integer id) {
        //根据自媒体文章ID 查询自媒体文章对象
        WmNews wmNews = wmNewsService.getById(id);
        audit(wmNews);
    }

    /**
     * 对提交的提审文章（自媒体文章）进行审核操作
     * 1.对提交的提审文章 文本，图片 审核
     * 2.远程调用文章微服务将文章进行保存-文章发表库
     *
     * @param wmNews
     */
    @Async //
    @Override
    public void audit(WmNews wmNews) {
        //1.提审文章状态：自动审核通过待发布状态/人工审核通过状态文章 - 判断是否到了发布时间 如果到了发布时间 调用发布文章方法完成发布
        //1.1 得到提审文章状态：根据文章发布时间是否大于当系统时间 决定是否要对文章进行发布
        Integer status = wmNews.getStatus();
        if (MediaEnum.NEWS_SUBMIT_STATUS_ARTIFICIAL_PASS.getVal().equals(status) || MediaEnum.NEWS_SUBMIT_STATUS_PASS.getVal().equals(status)) {
            if (System.currentTimeMillis() >= wmNews.getPublishTime().getTime()) {
                //1.2 远程调用文章微服务将文章进行发布 发布到文章库中
                remoteSaveArticle(wmNews);
            }
        }

        //2.提审文章状态为“待审核”，自动进入送审业务流程中  检查内容：文本（标题，正文）   图片（封面，正文图片）
        if (MediaEnum.NEWS_SUBMIT_STATUS_COMMIT.getVal().equals(status)) {
            //2.1 从提审自媒体文章中获取图片以及文本
            Map<String, Object> map = getTextAndImageFromContent(wmNews.getContent());


            //2.2 对获取到文本进行检测-更新文章状态
            String text = (String) map.get("text") + wmNews.getTitle();

            //2.2.1 判断自研敏感词是否存在
            boolean b = checkSensitive(text, wmNews);
            if (!b) {
                return;
            }

            boolean textResult = checkText(text, wmNews);
            if (!textResult) {
                return;
            }
            //2.3 对获取到图片进行检测--更新文章状态
            List<String> images = (List<String>) map.get("images");
            if (StringUtils.isNotBlank(wmNews.getImages())) {
                //处理封面中图片
                String[] split = wmNews.getImages().split(",");
                for (String s : split) {
                    images.add(s);
                }
            }

            boolean imageResult = checkImage(images, wmNews);
            if (!imageResult) {
                return;
            }

            //2.4 审核自媒体文章（两种情况：1未到发布时间，修改状态  2已经到发布时间：直接发布）
            //判断逻辑：提审文章发布时间跟当前时间比较 如果发布时间大于当前系统时间。不是直接发布，而是需要到了定义发布时间再次发布
            if (wmNews.getPublishTime() == null || wmNews.getPublishTime().getTime() <= System.currentTimeMillis()) {
                //2.4.1 直接发布
                remoteSaveArticle(wmNews);
            } else {
                //2.4.2 自媒体文章审核通过，但是没有到发布时间 TODO：基于延迟队列实现该业务功能 延迟发布
                wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_PASS.getVal());
                wmNewsService.updateById(wmNews);

                //2.4.3 TODO 生产者：发送延迟消息  延迟发布文章
                //计算延迟时间
                long time = wmNews.getPublishTime().getTime();
                long delayTime = time - System.currentTimeMillis();
                RBlockingDeque<Object> blockingDeque = redissonClient.getBlockingDeque("article-publish");
                RDelayedQueue<Object> delayedQueue = redissonClient.getDelayedQueue(blockingDeque);
                //发送延迟消息  自媒体文章ID
                delayedQueue.offer(wmNews.getId().toString(), delayTime, TimeUnit.MILLISECONDS);
            }
        }
    }

    /**
     * 对提审自媒体文章正文中：获取图片以及文本
     * [{type:"text",value:"第一段文本"},{type:"image",value:"图片地址"}]
     *
     * @param content
     * @return
     */
    private Map<String, Object> getTextAndImageFromContent(String content) {
        //0.定义文本结果
        StringBuilder stringBuilder = new StringBuilder();
        //0.1 定义图片集合
        List<String> images = new ArrayList<>();
        Map<String, Object> mapResult = new HashMap<>();
        //1.将内容转为JSON数组
        List<ContentDto> contentDtos = JSON.parseArray(content, ContentDto.class);
        //2.遍历数组对象 获取文本以及图片
        for (ContentDto dto : contentDtos) {
            if ("text".equals(dto.getType())) {
                stringBuilder.append(dto.getValue());
            } else {
                images.add(dto.getValue());
            }
        }
        mapResult.put("text", stringBuilder.toString());
        mapResult.put("images", images);
        return mapResult;
    }

    /**
     * 远程调用文章微服务-发布文章到发布库中
     *
     * @param wmNews 提审字体媒体文章
     */
    private void remoteSaveArticle(WmNews wmNews) {
        //1.构建发布文章实体类对象  ArticleDto:包含文章信息 以及 内容
        ArticleDto articleDto = new ArticleDto();
        //1.1 封装基本信息 将提审自媒体文章拷贝到 DTO中
        BeanUtils.copyProperties(wmNews, articleDto);
        //1.2 设置文章作者信息
        //1.2.1 根据自媒体用户ID查询自媒体用户对象
        Integer userId = wmNews.getUserId();
        WmUser wmUser = wmUserService.getById(userId);
        //1.2.2 根据自媒体用户得到作者信息
        articleDto.setAuthorId(wmUser.getApAuthorId());
        articleDto.setAuthorName(wmUser.getName());

        //1.3封装频道信息
        articleDto.setChannelId(wmNews.getChannelId());
        WmChannel channel = channelService.getById(wmNews.getChannelId());
        articleDto.setChannelName(channel.getName());

        //1.4 设置文章布局
        articleDto.setLayout(wmNews.getType());
        //1.5 设置文章标记 普通文章
        articleDto.setFlag(MediaEnum.ARTICLE_FLAG_NORMAL.getVal());

        //2. 远程调用文章微服务-发布文章到发布库中 -得到发布后文章ID
        ResponseResult<Long> responseResult = articleFeign.saveArticle(articleDto);
        if (responseResult.getCode().equals(AppHttpCodeEnum.SUCCESS.getCode())) {
            Long articleId = responseResult.getData();
            //3. 更新提审文章记录中 逻辑外键：文章ID 以及 状态改为：已发布
            wmNews.setArticleId(articleId);
            wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_PUBLISH.getVal());
            wmNewsService.updateById(wmNews);
        }
    }


    /**
     * 阿里云图片审核:审核自媒体文章对象中：封面，正文中图片
     *
     * @param images 封面+正文中图片
     * @param wmNews 自媒体文章对象
     * @return
     */
    private boolean checkImage(List<String> images, WmNews wmNews) {
        boolean result = false;
        // 如果没有图片,不需要进行图片审核,直接返回true
        if (images.size() <= 0) {
            return true;
        }
        try {
            // 调用阿里云图片审核
            // Map map = imageScan.checkUrl(images);    // 如果minio部署在外网服务器,可以直接通过URL地址来审核
            // 先将图片从minio服务器下载下来,然后调用阿里云进行审核
            List<byte[]> imageList = new ArrayList<>();
            for (String url : images) {
                // 下载图片 TODO:下载图片来源地址 图片都是从MinIO来的
                InputStream inputStream = minIOService.download(url);
                byte[] bytes = IOUtils.toByteArray(inputStream);
                imageList.add(bytes);
            }
            Map map = aliyunGreenService.imageScan(imageList);
            String suggestion = (String) map.get("suggestion");
            // 判断结果
            switch (suggestion) {
                case "pass":
                    // 审核通过
                    result = true;
                    break;
                case "block":
                    // 审核不通过
                    wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_FAIL.getVal());
                    // 获取不通过的原因
                    String label = (String) map.get("label");
                    wmNews.setReason("阿里云图片审核不通过: " + label);
                    wmNewsService.updateById(wmNews);
                    break;
                case "review":
                    // 人工审核
                    wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_ARTIFICIAL.getVal());
                    wmNewsService.updateById(wmNews);
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            throw new RuntimeException("审核图片失败");
        }
        return result;
    }

    /**
     * 阿里云文本审核
     *
     * @param text   文章文本
     * @param wmNews 自媒体文章对象
     * @return
     */
    private boolean checkText(String text, WmNews wmNews) {
        boolean result = false;
        try {
            Map map = aliyunGreenService.greenTextScan(text);
            String suggestion = (String) map.get("suggestion");
            // 判断结果
            switch (suggestion) {
                case "pass":
                    // 审核通过
                    result = true;
                    break;
                case "block":
                    // 审核不通过
                    wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_FAIL.getVal());
                    // 获取不通过的原因
                    String label = (String) map.get("label");
                    wmNews.setReason("阿里云文本审核不通过: " + label);
                    wmNewsService.updateById(wmNews);
                    break;
                case "review":
                    // 人工审核
                    wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_ARTIFICIAL.getVal());
                    wmNewsService.updateById(wmNews);
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            throw new RuntimeException("审核内容失败");
        }
        return result;
    }


    @Autowired
    private IWmSensitiveService sensitiveService;


    /**
     * 检查传入的文本是否包含敏感词
     *
     * @param text   文本
     * @param wmNews 自媒体用户发布的文章
     * @return
     */
    private boolean checkSensitive(String text, WmNews wmNews) {
        boolean result = false;
        //1.判断敏感词工具中的map是否为空，如果为空，从数据库中加载敏感词到Map中
        if (CollectionUtils.isEmpty(SensitiveWordUtil.dictionaryMap)) {
            LambdaQueryWrapper<WmSensitive> queryWrapper = new LambdaQueryWrapper<>();
            queryWrapper.select(WmSensitive::getSensitives);
            //1.1 查询敏感词表中名称列
            List<WmSensitive> sensitiveList = sensitiveService.list(queryWrapper);
            if (!CollectionUtils.isEmpty(sensitiveList)) {
                List<String> collect = sensitiveList.stream().map(WmSensitive::getSensitives).collect(Collectors.toList());
                //1.2 初始化敏感词map
                SensitiveWordUtil.initMap(collect);
            }
        }
        //2. 审核标题
        Map<String, Integer> mapTitle = SensitiveWordUtil.matchWords(wmNews.getTitle());
        if (mapTitle.size() <= 0) {
            //2.1 文章标题审核通过
            //2.1.1 继续审核文章内容是否包含敏感词
            Map<String, Integer> mapContent = SensitiveWordUtil.matchWords(text);
            if (CollectionUtils.isEmpty(mapContent)) {
                //内容审核不包含敏感词
                result = true;
            } else {
                //2.1.3 内容审核失败
                wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_FAIL.getVal());
                Set<String> keySet = mapContent.keySet();
                String join = String.join(",", keySet);
                wmNews.setReason("内容中包含敏感词：" + join);
                wmNewsService.updateById(wmNews);
            }
        } else {
            //2.2 文章标题审核失败
            wmNews.setStatus(MediaEnum.NEWS_SUBMIT_STATUS_FAIL.getVal());
            Set<String> keySet = mapTitle.keySet();
            String join = String.join(",", keySet);
            wmNews.setReason("标题中包含敏感词：" + join);
            wmNewsService.updateById(wmNews);
            return false;
        }
        return result;
    }
}
